package com.yww.leetcode.binarysearchtree;

import com.yww.leetcode.common.TreeNode;

import java.util.Random;

/**
 * @author yww
 * @description 701. 二叉搜索树中的插入操作
 * @since 2024/2/15 14:58
 */
public class InsertIntoBST701 {
    // 给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。
    // 返回插入后二叉搜索树的根节点。
    // 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

    // 注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。
    // 你可以返回 任意有效的结果 。
    public TreeNode insertIntoBST(TreeNode root, int val) {
        int i = new Random().nextInt() & 1;
        if (i == 0) {
            return recurInsert(root, val);
        }
        return doInsert(root, val);
    }

    /**
     * 递归插入
     */
    public TreeNode recurInsert(TreeNode cur, int val) {
        if (cur == null) {
            return new TreeNode(val, null, null);
        }
        if (val < cur.val) {
            cur.left = recurInsert(cur.left, val);
        } else {
            cur.right = recurInsert(cur.right, val);
        }
        return cur;
    }

    /**
     * 循环新增
     */
    public TreeNode doInsert(TreeNode root, int val) {
        TreeNode p = root;
        TreeNode parent = null;
        while (p != null) {
            if (val < p.val) {
                parent = p;
                p = p.left;
            } else {
                parent = p;
                p = p.right;
            }
        }
        if (parent == null) {
            return new TreeNode(val, null, null);
        } else if (val < parent.val) {
            parent.left = new TreeNode(val, null, null);
        } else {
            parent.right = new TreeNode(val, null, null);
        }
        return root;
    }
}
